/* Copyright (c) 2017, NVIDIA CORPORATION.  All rights reserved.
*
* NVIDIA CORPORATION and its licensors retain all intellectual property
* and proprietary rights in and to this software, related documentation
* and any modifications thereto.  Any use, reproduction, disclosure or
* distribution of this software and related documentation without an express
* license agreement from NVIDIA CORPORATION is strictly prohibited.
*/

#pragma once

#include "NVIDIAGfeSDKPrivatePCH.h"

#include <map>

extern std::map<GfeSDK::NVGSDK_Scope, EGfeSDKScope> kScopeSdkToUnreal;
extern std::map<EGfeSDKScope, GfeSDK::NVGSDK_Scope> kScopeUnrealToSdk;
extern std::map<GfeSDK::NVGSDK_Permission, EGfeSDKPermission> kPermissionSdkToUnreal;
extern std::map<EGfeSDKPermission, GfeSDK::NVGSDK_Permission> kPermissionUnrealToSdk;
extern std::map<GfeSDK::NVGSDK_HighlightType, EGfeSDKHighlightType> kHighlightTypeSdkToUnreal;
extern std::map<EGfeSDKHighlightType, GfeSDK::NVGSDK_HighlightType> kHighlightTypeUnrealToSdk;
extern std::map<GfeSDK::NVGSDK_HighlightSignificance, EGfeSDKHighlightSignificance> kHighlightSignificanceSdkToUnreal;
extern std::map<EGfeSDKHighlightSignificance, GfeSDK::NVGSDK_HighlightSignificance> kHighlightSignificanceUnrealToSdk;

extern EGfeSDKReturnCode TranslateReturnCodeToUnreal(GfeSDK::NVGSDK_RetCode code);

template <typename T, typename U>
std::map<U, T> InvertMap(std::map<T, U> const& map)
{
    std::map<U, T> result;
    for (auto it = map.begin(); it != map.end(); ++it)
    {
        result[it->second] = it->first;
    }
    return result;
}

template <typename T, typename U>
U TranslateEnum(std::map<T, U> const& map, T const& value, U const& defaultValue)
{
    auto foundIt = map.find(value);
    return (foundIt != map.end()) ? foundIt->second : defaultValue;
}

template <typename T, typename U>
U TranslateBitfieldEnum(std::map<T, U> const& map, T const& value)
{
    uint32 result = 0x0;
    for (size_t i = 0; i < sizeof(T)*8; ++i)
    {
        uint32 v = (1 << i);
        if ((static_cast<uint32>(value) & v) > 0)
        {
            auto foundIt = map.find(static_cast<T>(v));
            if (foundIt != map.end())
            {
                result |= static_cast<uint32>(foundIt->second);
            }
        }
    }

    return static_cast<U>(result);
}
